import paddlehub as hub

module = hub.Module(name="ernie_gen_couplet")

test_texts = ["长河落日圆", "零落成泥捻作尘"]
# generate包含3个参数，texts为输入文本列表，use_gpu指定是否使用gpu，beam_width指定beam search宽度。
results = module.generate(texts=test_texts, use_gpu=True, beam_width=5)
for result in results:
    print(result)